//  https://leetcode-cn.com/problems/wiggle-subsequence
//  WiggleSubsequence.swift
//  LeetCodeSwift
//
//  Created by XFB on 2019/8/2.
//  Copyright © 2019 XFB. All rights reserved.
//  摇摆序列

import Foundation

class WiggleSubsequence {
   
    func wiggleMaxLength(_ nums: [Int]) -> Int {

        /**
         如果连续数字之间的差严格地在正数和负数之间交替，则数字序列称为摆动序列。第一个差（如果存在的话）可能是正数或负数。少于两个元素的序列也是摆动序列。
         例如， [1,7,4,9,2,5] 是一个摆动序列，因为差值 (6,-3,5,-7,3) 是正负交替出现的。相反, [1,4,7,2,5] 和 [1,7,4,5,5] 不是摆动序列，第一个序列是因为它的前两个差值都是正数，第二个序列是因为它的最后一个差值为零。
         
         给定一个整数序列，返回作为摆动序列的最长子序列的长度。 通过从原始序列中删除一些（也可以不删除）元素来获得子序列，剩下的元素保持其原始顺序。
         
         示例 1:
         输入: [1,7,4,9,2,5]
         输出: 6
         解释: 整个序列均为摆动序列。
         
         示例 2:
         输入: [1,17,5,10,13,15,10,5,16,8]
         输出: 7
         解释: 这个序列包含几个长度为 7 摆动序列，其中一个可为[1,17,10,13,10,16,8]。
         
         示例 3:
         输入: [1,2,3,4,5,6,7,8,9]
         输出: 2
         */
        
        /// 序列个数小于2时 直接为摇摆序列
        if nums.count < 2 {
            return nums.count
        }
        
        /// 扫描序列的三种状态
        let BEGIN = 0
        let UP = 1
        let DOWN = 2
    
        var STARTE = BEGIN
        /// 摇摆序列最大长度至少为 1
        var max_length = 1
    
        /// 从第二个元素开始扫描
        for i in 1 ..< nums.count {
            
            switch STARTE {
            case BEGIN:
                if (nums[i - 1] < nums[i]) {
                    STARTE = UP
                    max_length = max_length + 1
                } else if (nums[i - 1] > nums[i]) {
                    STARTE = DOWN
                    max_length = max_length + 1
                }
                break
            case UP:
                if (nums[i - 1] > nums[i]) {
                    STARTE = DOWN
                    max_length = max_length + 1
                }
                break
            case DOWN:
                if (nums[i - 1] < nums[i]) {
                    STARTE = UP
                    max_length = max_length + 1
                }
                break
            default:
                break
            }
        }
        return max_length
    }
}
